package com.ysh.smartplatform.controller.advice;

import com.fasterxml.jackson.core.JsonProcessingException;
import com.fasterxml.jackson.databind.ObjectMapper;
import com.ysh.smartplatform.constans.ResultCode;
import com.ysh.smartplatform.dto.R;
import com.ysh.smartplatform.exceptions.MyBaseException;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.MethodArgumentNotValidException;
import org.springframework.web.bind.annotation.ExceptionHandler;
import org.springframework.web.bind.annotation.RestController;
import org.springframework.web.bind.annotation.RestControllerAdvice;

import java.util.Map;

/**
 * 袁胜虎 on 2022/9/5 21:29
 */

/**
 * 通过@ControllerAdvice注解可以将对于控制器的全局配置放在同一个位置。
 * 注解了@RestControllerAdvice的类的方法可以使用@ExceptionHandler、@InitBinder、@ModelAttribute注解到方法上。
 * @RestControllerAdvice注解将作用在所有注解了@RequestMapping的控制器的方法上。
 * @ExceptionHandler：用于指定异常处理方法。当与@RestControllerAdvice配合使用时，用于全局处理控制器里的异常。
 * @InitBinder：用来设置WebDataBinder，用于自动绑定前台请求参数到Model中。
 * @ModelAttribute：本来作用是绑定键值对到Model中，当与@ControllerAdvice配合使用时，可以让全局的@RequestMapping都能获得在此处设置的键值对
 * ————————————————
 * 版权声明：本文为CSDN博主「user2025」的原创文章，遵循CC 4.0 BY-SA版权协议，转载请附上原文出处链接及本声明。
 * 原文链接：https://blog.csdn.net/user2025/article/details/105458842
 */
@RestControllerAdvice
public class MyControllerAdvice {
    private ObjectMapper objectMapper;

    @Autowired
    public void setObjectMapper(ObjectMapper objectMapper) {
        this.objectMapper = objectMapper;
    }
    /**
     * 处理我们的自定义异常的
     *
     * @param exception
     * @return
     */
    @ExceptionHandler(MyBaseException.class)
    public R processBaseException(MyBaseException exception){
        exception.printStackTrace();
        return R.getResult(exception.getCode(),exception.getMessage());
    }
    /**
     * 处理请求参数校验的异常
     * @param exception
     * @return
     */
    @ExceptionHandler(MethodArgumentNotValidException.class)
    public R processMethodArgumentNotValidException(MethodArgumentNotValidException exception) {
        String defaultMessage = exception.getBindingResult().getAllErrors().get(0).getDefaultMessage();
        try {
            //将我们自定义的异常字符串转成map,然后从里面获取到我们传递的code和msg
            Map map = objectMapper.readValue(defaultMessage, Map.class);
            return R.getResult((Integer) map.get("code"), (String) map.get("msg"));
        } catch (JsonProcessingException e) {
            e.printStackTrace();
        }
        return R.ERROR();
    }

    @ExceptionHandler(Exception.class)
    public R processBaseException(Exception e){
        e.printStackTrace();
        return R.getResult(ResultCode.FAIL,"网络异常,请见检查您的网络");
    }
}
